import os

def check_project_structure():
    """检查项目结构"""
    print("检查项目结构...")
    
    # 打印当前工作目录
    current_dir = os.getcwd()
    print(f"当前工作目录: {current_dir}")
    
    # 需要检查的目录和文件
    required_structure = {
        'src': ['main.py', 'data_processor.py', 'model_builder.py', 'trainer.py'],
        'dataset': [],
        'models': [],
        'logs': []
    }
    
    # 检查每个目录
    for directory, files in required_structure.items():
        dir_path = os.path.join(current_dir, directory)
        if not os.path.exists(dir_path):
            print(f"❌ 目录不存在: {dir_path}")
            continue
        print(f"✅ 找到目录: {directory}")
        
        # 检查文件
        for file in files:
            file_path = os.path.join(dir_path, file)
            if os.path.exists(file_path):
                print(f"  ✅ 找到文件: {file}")
                # 检查文件大小
                size = os.path.getsize(file_path)
                print(f"     文件大小: {size} 字节")
            else:
                print(f"  ❌ 文件不存在: {file}")
    
    # 列出当前目录下的所有文件和文件夹
    print("\n当前目录下的所有文件和文件夹:")
    for item in os.listdir(current_dir):
        item_path = os.path.join(current_dir, item)
        if os.path.isdir(item_path):
            print(f"📁 {item}/")
        else:
            print(f"📄 {item}")

if __name__ == "__main__":
    check_project_structure() 